package com.librelato.tela.datamodel;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import com.librelato.bean.UnidadeRequisicaoBean;

public class UnidadeDataModel extends ListDataModel<UnidadeRequisicaoBean>
		implements SelectableDataModel<UnidadeRequisicaoBean> {
	
	
	public UnidadeDataModel(List<UnidadeRequisicaoBean> data) {
		super(data);
	}
	

	@SuppressWarnings("unchecked")
	@Override
	public UnidadeRequisicaoBean getRowData(String linha) {
		List<UnidadeRequisicaoBean> listaUni = (List<UnidadeRequisicaoBean>) getWrappedData();
		for(UnidadeRequisicaoBean uni : listaUni){
			if(uni.getCodigo().equals(linha))
				return uni;
		}
		
		return null;
	}

	@Override
	public Object getRowKey(UnidadeRequisicaoBean linha) {
		return linha.getCodigo();
	}

}
